package com.rr.middle;

import java.util.*;

/**
 * https://leetcode.cn/problems/permutations-ii/description/
 */
public class PermuteUnique {
    /**
     * 个人解：有点暴力解法了，用set辅助去重
     */

//    private Set<List<Integer>> resultSet = new HashSet<>();
//
//    public List<List<Integer>> permuteUnique(int[] nums) {
//        Set<Integer> steps = new HashSet<>();
//        for (int i = 0; i < nums.length; i++) {
//            steps.add(i);
//        }
//        buildResult(Collections.emptyList(), steps, nums);
//        return new ArrayList<>(resultSet);
//    }
//
//    private void buildResult(List<Integer> list, Set<Integer> newSteps, int[] nums) {
//        if (newSteps.size() == 0) {
//            resultSet.add(list);
//            return;
//        }
//        for (Integer newStep : newSteps) {
//            HashSet<Integer> nextSets = new HashSet<>(newSteps);
//            nextSets.remove(newStep);
//            List<Integer> nextList = new ArrayList<>(list);
//            nextList.add(nums[newStep]);
//            buildResult(nextList, nextSets, nums);
//        }
//    }

    /**
     * 官方解：回溯法
     */
    boolean[] vis;

    public List<List<Integer>> permuteUnique(int[] nums) {
        List<List<Integer>> ans = new ArrayList<List<Integer>>();
        List<Integer> perm = new ArrayList<Integer>();
        vis = new boolean[nums.length];
        Arrays.sort(nums);
        backtrack(nums, ans, 0, perm);
        return ans;
    }

    public void backtrack(int[] nums, List<List<Integer>> ans, int idx, List<Integer> perm) {
        if (idx == nums.length) {
            ans.add(new ArrayList<Integer>(perm));
            return;
        }
        for (int i = 0; i < nums.length; ++i) {
            // 重点判断：如果当前结点没放进，该结点又跟上一节点值相同，并且上一个结点已经是false(说明由它来回溯的结果都生成了)，此时就不需要再对该结点做回溯了
            if (vis[i] || (i > 0 && nums[i] == nums[i - 1] && !vis[i - 1])) {
                continue;
            }
            perm.add(nums[i]);
            vis[i] = true;
            backtrack(nums, ans, idx + 1, perm);
            vis[i] = false;
            perm.remove(idx);
        }
    }

    public static void main(String[] args) {
        PermuteUnique permuteUnique = new PermuteUnique();
        System.out.println(permuteUnique.permuteUnique(new int[] {1, 1, 2}));
    }
}
